"""
实现功能：
    批量生成文件夹
"""
import os
import openpyxl
from openpyxl import Workbook

# 用户输入目标文件夹路径
folder_path = 'E:/02_item/02_学位申请/05_2025春/01_学位预申请材料'

# 定义需要提取的文件类型
desired_file_type = ".xlsx"

# 遍历所有目录下的指定文件格式的路径
def get_files(dir, suffix):
    py_files = []
    for root, dirs, files in os.walk(dir):  # 遍历所有目录，包括自身
        for file in files:  # 遍历文件，抓取指定文件
            pre, suf = os.path.splitext(file)
            if suf == suffix:
                py_files.append(os.path.join(root, file))
    return py_files


if __name__ == '__main__':
    # 生成新的excel文档
    new_workbook = Workbook()
    new_sheet = new_workbook.active

    for i in get_files(folder_path, desired_file_type):
        workbook = openpyxl.load_workbook(i)
        worksheet = workbook.active

        # 输出第三行内容
        if worksheet.max_row >= 3:  # 确保表格至少有3行
            third_row = []
            for cell in worksheet[3]:  # 通过行号获取第三行的所有单元格
                third_row.append(cell.value)
        else:
            print("表格中没有足够的行数")

        new_sheet.append(third_row)
        # 本地保存test.xlsx
        new_workbook.save("E:/02_item/02_学位申请/03_2024春/01_学位预申请材料/1.xlsx")
